# -*- coding: utf-8 -*-
"""
Created on Tue Oct 25 14:03:56 2016

@author: Robin van Emden

Dataset:

	tcont: t for duration of experiment
	t: t per phase
	x0_a1: x0 of attribute a1
	x0_a2: x0 of attribute a2
	x_a1: x of attribute a1
	x_a2: x of attribute a2
	y: y (response) 
	phase: phase of the experiment 

Demographics:
	
	where_do_you_currently_reside
	whats_your_gender
	whats_your_age
	what_is_the_highest_level_of_education_you_have_received

"""

import matplotlib.pyplot as plt
import pandas as pd

input = pd.read_csv("face_clean_with_demo.csv");

ra = pd.rolling_mean(input.y, 150)

plt.figure(figsize=(8, 3))
plt.plot(input.tcont,input.x0_a1, label='x0_a1')
plt.xlabel('t')
plt.ylabel('')
plt.ylim([10,100])
plt.legend()
plt.show()

plt.figure(figsize=(8, 3))
plt.plot(input.tcont,input.x0_a2, label='x0_a2')
plt.xlabel('t')
plt.ylabel('')
plt.ylim([10,70])
plt.legend()
plt.show()

plt.figure(figsize=(8, 3))
plt.plot(input.tcont,input.y, label='y')
plt.plot(input.tcont,ra, label='rolling average (150)')
plt.xlabel('t')
plt.ylabel('')
plt.legend()
plt.show()